package club.xiaojiawei.array;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 2022/4/17 21:58
 * @question 27.移除元素
 * @description 给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
 */
public class RemovingElements27 {
    public static void main(String[] args) {
        int[] arr={3,2,2,3};
        int remove=1;
        int count=0;
        for (int i1 = 0; i1 < arr.length; i1++) {
            if (arr[i1]==remove){
                count++;
            }
        }
        System.out.println("正确答案："+(arr.length-count));
        int i = removeElement(arr, remove);
        System.out.println("实际答案："+i);
        for (int j = 0; j < i; j++) {
            if (arr[j] == remove){
                System.out.println("错误");
            }else {
                System.out.print(arr[j]+" , ");
            }
        }
    }

    /**
     * 双指针法
     * @param nums
     * @param val
     * @return
     */
    public static int removeElement(int[] nums, int val) {
        int rear=nums.length;
        for (int front = 0; front < rear; front++) {
            if (nums[front] == val){
//                首尾指针相遇，直接返回rear-1
                if (front == (rear - 1)){
                    return --rear;
                }
//                从尾部寻找非移除元素
                while (nums[rear-1] == val){
//                首尾指针相遇，直接返回rear-1
                    if (front == --rear){
                        return --rear;
                    }
                }
//                将front处要移除的元素赋值为尾部非移除元素
                nums[front] = nums[--rear];
            }
        }
        return rear;
    }
}
